package com.sweet.string;

import java.io.UnsupportedEncodingException;

/**
 * GB 2312-80 把收录的汉字分成两级。<br/>
 * 第一级汉字是常用汉字，计 3755 个， 置于 16～55区，按汉语拼音字母／笔形顺序排列；<br/>
 * 第二级汉字是次常用汉字， 计 3008 个，置于 56～87 区，按部首／笔画顺序排列；<br/>
 * 所以本程序只能查到 对一级汉字的声母。同时对符合声母（zh，ch，sh）只能取首字母（z，c，s）
 */
public final class ChineseToLetter {
    /**
     * 国标码和区位码转换常量
     */
    private static final int GB_SP_DIFF = 160;

    /**
     * 存放国标一级汉字不同读音的起始区位码
     */
    private static final int[] secPosValueList = {1601, 1637, 1833, 2078, 2274, 2302, 2433, 2594, 2787, 3106, 3212, 3472, 3635, 3722, 3730, 3858, 4027, 4086, 4390, 4558, 4684, 4925, 5249, 5600};

    /**
     * 存放国标一级汉字不同读音的起始区位码对应读音
     */
    private static final char[] firstLetter = {'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'w', 'x', 'y', 'z'};

    private ChineseToLetter() {
    }

    /**
     * 获取首字母
     *
     * @param bytes
     * @return
     */
    private static char convert(byte[] bytes) {
        char result = 'o';// 如果找不到对应的，则用o代替
        int secPosValue = 0;
        int i;
        for (i = 0; i < bytes.length; i++) {
            bytes[i] -= GB_SP_DIFF;
        }
        secPosValue = bytes[0] * 100 + bytes[1];
        for (i = 0; i < 23; i++) {
            if (secPosValue >= secPosValueList[i] && secPosValue < secPosValueList[i + 1]) {
                result = firstLetter[i];
                break;
            }
        }
        return result;
    }

    /**
     * 获取每个汉字的首字母
     *
     * @param oriStr
     * @return
     */
    public static String getFirstLetter(String oriStr) {
        try {
            String str = oriStr.toLowerCase();
            StringBuffer buffer = new StringBuffer();
            char ch;
            char[] temp;
            for (int i = 0; i < str.length(); i++) { // 依次处理str中每个字符
                ch = str.charAt(i);
                temp = new char[]{ch};

                // 必须使用GBK编码，因为在linux下，默认会使用utf-8导致汉字转为byte失败,获取的首字母全是0
                byte[] uniCode = new String(temp).getBytes("GBK");

                // 非汉字直接追加，汉字则获取首字母
                if (uniCode[0] < 128 && uniCode[0] > 0) {
                    buffer.append(temp);
                } else {
                    buffer.append(convert(uniCode));
                }
            }

            return buffer.toString();
        } catch (UnsupportedEncodingException e) {
            return "";
        }
    }
}
